import numpy as np
import os  # 导入os模块用于创建目录
from date_generation import generate_glv_communities  # 导入生成GLV的函数
# 生成数据（如果已有数据，直接加载即可）
num_samples = 10
N = 100
s, p = generate_glv_communities(num_samples=num_samples, N=N)
# 关键：创建data目录（如果不存在）
os.makedirs("data", exist_ok=True)  # exist_ok=True表示目录存在时不报错
# 保存到data目录
np.savez("data/glv_data.npz", s=s, p=p)
print("数据已保存到 data/glv_data.npz")